package com.indusfo.spc.controller;

import com.indusfo.spc.exception.GlobalException;
import com.indusfo.spc.exception.ModifyFailedException;
import com.indusfo.spc.exception.ParamsErrorException;
import com.indusfo.spc.exception.QueryFailedException;
import com.indusfo.spc.mapper.UploadMapper;
import com.indusfo.spc.pojo.*;
import com.indusfo.spc.service.ProductLineService;
import com.indusfo.spc.service.StationService;
import com.indusfo.spc.service.WorkshopService;
import com.indusfo.spc.untlis.GsonUtil;
import com.indusfo.spc.vo.CommonUtil;
import com.indusfo.spc.vo.JSONObject;
import com.indusfo.spc.vo.ZTreeNode;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;

/**
 * 工位前端控制层
 * @author chenm
 * @create 2019-07-13 14:40
 */
@Controller
@RequestMapping("/pstation")
public class StationController {

    @Autowired
    private StationService stationService;

    @Resource
    private WorkshopService workShopService;

    @Autowired
    private ProductLineService productLineService;

    @Resource
    private UploadMapper uploadMapper;

    /**
     * 新增
     * @param station
     * @return
     */
    @RequestMapping(value="/insertStation" ,method= RequestMethod.POST)
    @ResponseBody
    public JSONObject insertStation(Station station){
        JSONObject json = stationService.insertStation(station);
        return json;

    }

    /**
     * 删除
     * @param station
     * @return
     */
    @RequestMapping(value="/delStation", method= RequestMethod.POST)
    @ResponseBody
    public JSONObject delStation(Integer[] staIds, Integer dataState) {

        dataState = 2;
        JSONObject json = stationService.delStation(staIds, dataState);
        return json;
    }

    /**
     * 启用/停用
     *
     * @author xuz
     * @date 2018/12/21 2:52 PM
     * @param [lProductLineIds, lDataState]
     * @return com.indusfo.spc.vo.JSONObject
     */
    @RequestMapping(value="/definesStation", method= RequestMethod.POST)
    @ResponseBody
    public JSONObject definesStation(Integer[] staIds, Integer dataState) {

        JSONObject json = stationService.delStation(staIds, dataState);
        return json;
    }

    /**
     * 更改
     * @param station
     * @return
     */
    @RequestMapping(value="/updateStation", method= RequestMethod.POST)
    @ResponseBody
    public JSONObject updateStation(Station station) {

        JSONObject json = stationService.updateStation(station);
        return json;

    }

    /**
     * 查询
     * @param station
     * @return
     */
    @RequestMapping(value="/queryStation")
    @ResponseBody
    public JSONObject queryStation(Station station) {

        JSONObject json = stationService.queryStation(station);
        return json;

    }

    /**
     * 根据车间，产线，工序，查询工位集合
     * @param station
     * @return
     */
    @RequestMapping(value="/primary")
    @ResponseBody
    public JSONObject primary(Integer id, String name) {
        List<Station> liststStations = null;
        try {
            if("production".equals(name)){
                liststStations = stationService.getByProductionLineId(id);
            }else if(name.contentEquals("workshop")){
                liststStations = stationService.getByWorkshopId(id);
            }else if("pro".equals(name)){
                liststStations = stationService.getByProId(id);
            }else{
                throw new QueryFailedException("输入参数有误");
            }

            if (liststStations.isEmpty()) {
                throw new QueryFailedException("没有产线相关数据");
            }
            //查询分页总记录数
            return JSONObject.oK(liststStations);
        } catch (GlobalException e) {
            return JSONObject.build(403, e.getMessage());
        }

    }

    /**
     * 车间，产线，工位三级树形结构
     * @param station
     * @return
     */
    @RequestMapping(value = "/json/Tree", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public JSONObject queryMouldTree() {
        try {
            List<ZTreeNode> nodes = new ArrayList<>();
            List<Workshop> workShops = workShopService.getList();
            List<Productline> productLines = new ArrayList<Productline>();
            if (null != workShops) {
                for (Workshop _workShop : workShops) {
                    ZTreeNode node = new ZTreeNode();
                    node.setId(_workShop.getWorkshopId()+"W");
                    node.setPId(null);
                    node.setOpen(true);
                    node.setFlag("chejian");
                    node.setTitle(_workShop.getWorkshopName());
                    nodes.add(node);

                    List<Productline> lines = productLineService.getByWorkshopId(_workShop.getWorkshopId());
                    productLines.addAll(lines);
                }
            }
            List<Station> stations = new ArrayList<Station>();
            for(Productline _Line:productLines){
                ZTreeNode node = new ZTreeNode();
                node.setId(_Line.getProductlineId()+"P");
                node.setPId(_Line.getWorkshopId()+"W");
                node.setOpen(true);
                node.setFlag("canxian");
                node.setTitle(_Line.getProductlineName());
                nodes.add(node);
                List<Station> list = stationService.getByProductionLineId(_Line.getProductlineId());
                stations.addAll(list);
            }
            for(Station _Station:stations){
                ZTreeNode node = new ZTreeNode();
                node.setId(_Station.getStationId()+"S");
                node.setPId(_Station.getProductlineId()+"P");
                node.setOpen(true);
                node.setFlag("gongwei");
                node.setTitle(_Station.getStationName());
                nodes.add(node);
            }
            List<ZTreeNode> tree = CommonUtil.getTree(nodes, null);
            return JSONObject.oK(tree);
        } catch (ParamsErrorException e) {
            //返回封装异常信息的JSONObject给前端
            return JSONObject.build(404, e.getMessage());
        } catch (ModifyFailedException e) {
            return JSONObject.build(404, e.getMessage());
        }
    }
    /**
     * @author: louk
     * @Description:删除工序信息下的工位
     * @date: 2019/9/5 14:24
     *
     */
    @RequestMapping(value = "/delStationByPro")
    @ResponseBody
    public JSONObject delStationByPro(Long[] staIds) {

        System.out.println(staIds);
        return stationService.delStationByPro(staIds);
    }
    @RequestMapping("/uploadFileStation")
    @ResponseBody
    public JSONObject uploadFileUser(@RequestParam("file") MultipartFile file, Boolean way, HttpServletRequest request, HttpServletResponse response) throws IOException, ParseException {
        //定义一个message对象用于向前台传输数据
        Message msg = new Message();
        //设置错误集合的
        List<ExcelErrorData> errorDatas = new ArrayList<>();
        response.setContentType("text/html;charset=UTF-8");
        //判断上传文件格式是否正确
        Workbook wb = ExcelUtil.getWorkbook(file.getInputStream(), ExcelUtil.isExcel2007(file.getOriginalFilename()));
        if (wb == null) {
            errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "文件解析错误"));
        } else {
            //获取excel文件的第一个sheet
            Sheet sheet = wb.getSheetAt(0);
            if (sheet == null) {
                msg.setCode(2);
                msg.setMsg("未找到需要导入的表格");
                errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "未找到需要导入的表格"));
            } else {
                //获取最后一行的行号
                int rowCount = sheet.getLastRowNum();
                if (rowCount < 1) {
                    msg.setCode(1);
                    msg.setMsg("未找到需要导入的数据，请确认表格是否保存！！！");
                    errorDatas.add(ExcelErrorData.getExcelErrorData(0, "未找到需要导入的数据，请确认表格是否保存！！！"));
                } else {
                    List<Station> addProductPlans = new ArrayList<>();
                    for (int i = 1; i <= rowCount; i++) {
                        boolean needAdd = true;
                        //获取到每一行的每一列的数据
                        String staionCode =com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 0, false);
                        String staionName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 1, false);
                        String staType = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 2, false);
                        String productionLineName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 3, false);
                        String remark = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 4, false);
                        Station plan = new Station(staionCode,staionName,staType,productionLineName,remark);
                        int m = i + 1;
                        //将excel中的每一行数据存储进行检查
                        String result = stationService.checkExcelContent(plan);
                        if(result!=null){
                            errorDatas.add(ExcelErrorData.getExcelErrorData(i, result));
                            needAdd=false;
                        }
                        for (int j = 1; j < i; j++) {
                            //判断用户编码是否重复
                            String oldStaionCode =com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, j, 0, false);
                            String oldStaionName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, j, 1, false);
                            if (StringUtils.isBlank(staionCode) && oldStaionCode.equals(staionCode)) {
                                needAdd = false;
                                errorDatas.add( ExcelErrorData.getExcelErrorData(i, "与表格中第" + j + "行存在相同工位编码"));
                            }
                            if (StringUtils.isBlank(staionName) && oldStaionName.equals(staionName)) {
                                needAdd = false;
                                errorDatas.add( ExcelErrorData.getExcelErrorData(i, "与表格中第" + j + "行存在相同工位名称"));
                            }
                        }
                        if (needAdd) {
                            addProductPlans.add(plan);
                        } else {
                            msg.setCode(2);
                            msg.setMsg("导入数据中存在异常信息");
                        }

                    }
                    //根据这些来判断是否是中止导入还是跳过继续进行导入
                    if (msg.getCode() == 2 && !way) {
                        msg.setMsg(msg.getMsg() + ",中止导入");
                    } else {
                        if (addProductPlans.size() > 0) {
                            try {
                                uploadMapper.updateListStation(addProductPlans);
                                //当程序执行到这里时,判断信息是否为空为空的话设置导入成功
                                msg.setMsg(StringUtils.isBlank(msg.getMsg()) ? "导入成功" : msg.getMsg());
                                if ("导入成功".equals(msg.getMsg())) {
                                    msg.setCode(1);
                                }
                            } catch (Exception e) {
                                e.printStackTrace();
                                msg.setMsg("导入失败");
                            }
                        }
                    }

                }
            }
        }
        msg.setData(GsonUtil.toJson(errorDatas));
        return JSONObject.build(200,msg.getMsg(),errorDatas);
    }


}
